package com.ai.demo.invoke;

import jakarta.annotation.Resource;
import org.springframework.ai.chat.client.advisor.MessageChatMemoryAdvisor;
import org.springframework.ai.chat.client.advisor.api.Advisor;
import org.springframework.ai.chat.memory.ChatMemory;
import org.springframework.ai.chat.messages.AssistantMessage;
import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.chat.prompt.Prompt;
import org.springframework.boot.CommandLineRunner;
import org.springframework.stereotype.Component;

/**
 * Spring Ai 框架调用大模型
 * */
@Component
public class SpringAiInvoke implements CommandLineRunner {

    @Resource
    private ChatModel chatModel;


    @Override
    public void run(String... args) throws Exception {
        AssistantMessage output = chatModel.call(new Prompt("你好，我是szs"))
                .getResult()
                .getOutput();
        System.out.println(output);
    }



}
